<?php
//  From WordPress DB Class
//  ORIGINAL CODE FROM:
//  Justin Vincent (justin@visunet.ie)
//	http://php.justinvincent.com

	
class DB {

	var $show_errors = true;
	var $num_queries = 0;
	var $last_query;
	var $col_info;
	var $queries;
	var $prefix='';

	var $charset = 'utf8';
	var $collate;
	var $debug = false;

	/**
	 * construct
	 * @param $dbuser
	 * @param $dbpassword
	 * @param $dbname
	 * @param $dbhost
	 * @return void
	 */
	function DB($dbuser, $dbpassword, $dbname, $dbhost='localhost') {
		register_shutdown_function(array(&$this, "__destruct"));
		$this->dbh = @mysql_connect($dbhost, $dbuser, $dbpassword);
		if (!$this->dbh) {
			$this->bail("<h3>Connect database be fail</h3><p>Pealse inspect yours database setting!</p>");
		}
		if ( !empty($this->charset) && version_compare(mysql_get_server_info(), '4.1.0', '>=') ) {
			$this->query("SET NAMES '$this->charset'");
		}		
		
		$this->select($dbname);
	}

	/**
	 * Selects a database using the current class's $this->dbh
	 * @param string $db name
	 */
	function select($db) {
		if (!@mysql_select_db($db, $this->dbh)) {
			$this->bail("<h3>Can't select name is '". $db ."' database</h3><p>Pealse inspect yours database setting!</p>");
		}
	}

	/**
	 * Escapes content for insertion into the database, for security
	 * @param string $string
	 * @return string query safe string
	 */
	function escape($string) {
		return addslashes( $string ); // Disable rest for now, causing problems
		if( !$this->dbh || version_compare( phpversion(), '4.3.0' ) == '-1' )
			return mysql_escape_string( $string );
		else
			return mysql_real_escape_string( $string, $this->dbh );
	}

	/**
	 * Print SQL/DB error.
	 * @param $str
	 * @return boolean
	 */
	function print_error($str = '') {
		global $EZSQL_ERROR;
		if (!$str) $str = mysql_error($this->dbh);
		$EZSQL_ERROR[] =
		array ('query' => $this->last_query, 'error_str' => $str);

		$str = htmlspecialchars($str, ENT_QUOTES);
		$query = htmlspecialchars($this->last_query, ENT_QUOTES);
		// Is error output turned on or not..
		if ( $this->show_errors ) {
			// If there is an error then take note of it
			print "<div id='error'>
			<p><strong>Exceed database error:</strong> [$str]<br />
			<code>$query</code></p>
			</div>";
		} else {
			return false;
		}
	}

	/**
	 * Turn error handling on or off..
	 * @return unknown_type
	 */
	function show_errors() {
		$this->show_errors = true;
	}

	function hide_errors() {
		$this->show_errors = false;
	}

	/**
	 * Kill cached query results
	 * @return void
	 */
	function flush() {
		$this->last_result = array();
		$this->col_info = null;
		$this->last_query = null;
	}

	/**
	 * base query method
	 * @param $query
	 * @return int return manipulative record count
	 */
	function query($query) {
		// filter the query, if filters are available
		// NOTE: some queries are made before the plugins have been loaded, and thus cannot be filtered with this method
		$query = str_replace('#nb@__',$this->prefix, $query);
		if ( function_exists('apply_filters') )
			$query = apply_filters('query', $query);

		// initialise return
		$return_val = 0;
		$this->flush();

		// Log how the function was called
		$this->func_call = "\$db->query(\"$query\")";

		// Keep track of the last query for debug..
		$this->last_query = $query;
		
		$this->result = @mysql_query($query, $this->dbh);
		
		if($this->debug){
			log_debug( "executed sql: ". $query );
		}
		
		++$this->num_queries;

		// If there is an error then take note of it..
		if ( mysql_error($this->dbh) ) {
			$this->print_error();
			return false;
		}

		if ( preg_match("/^\\s*(insert|delete|update|replace) /i",$query) ) {
			$this->rows_affected = mysql_affected_rows($this->dbh);
			// Take note of the insert_id
			if ( preg_match("/^\\s*(insert|replace) /i",$query) ) {
				$this->insert_id = mysql_insert_id($this->dbh);
			}
			// Return number of rows affected
			$return_val = $this->rows_affected;
		} else {
			$i = 0;
			while ($i < @mysql_num_fields($this->result)) {
				$this->col_info[$i] = @mysql_fetch_field($this->result);
				$i++;
			}
			$num_rows = 0;
			while ( $row = @mysql_fetch_object($this->result) ) {
				$this->last_result[$num_rows] = $row;
				$num_rows++;
			}

			@mysql_free_result($this->result);

			// Log number of rows the query returned
			$this->num_rows = $num_rows;

			// Return number of rows selected
			$return_val = $this->num_rows;
		}

		return $return_val;
	}

	/**
	 * Get one variable from the database
	 * @param string $query (can be null as well, for caching, see codex)
	 * @param int $x = 0 row num to return
	 * @param int $y = 0 col num to return
	 * @return mixed results
	 */
	function get_var($query=null, $x = 0, $y = 0) {
		$this->func_call = "\$db->get_var(\"$query\",$x,$y)";
		if ( $query )
			$this->query($query);

		// Extract var out of cached results based x,y vals
		if ( $this->last_result[$y] ) {
			$values = array_values(get_object_vars($this->last_result[$y]));
		}
		// If there is a value return it else return null
		return (isset($values[$x]) && $values[$x]!=='') ? $values[$x] : null;
	}
	
	/**
	 * Return an entire result set from the database
	 * @param string $query (can also be null to pull from the cache)
	 * @return mixed results
	 */
	function get_results($query = null) {
		$this->func_call = "\$db->get_results(\"$query\", $output)";
		if ( $query ) {
			$this->query($query);
		}
			
		if ( $this->last_result ) {
			$i = 0;
			foreach( $this->last_result as $row ) {
				$new_array[$i] = (array) $row;
				/**
				if ( $output == ARRAY_N ) {
					$new_array[$i] = array_values($new_array[$i]);
				}**/
				$i++;
			}
			return $new_array;
		} else {
			return null;
		}	
	}
	
	function get_result($query = null) {
		$rs = $this->get_results($query);
		return array_pop($rs);
	}

	/**
	 * Starts the timer, for debugging purposes
	 */
	function timer_start() {
		$mtime = microtime();
		$mtime = explode(' ', $mtime);
		$this->time_start = $mtime[1] + $mtime[0];
		return true;
	}

	/**
	 * Stops the debugging timer
	 * @return int total time spent on the query, in milliseconds
	 */
	function timer_stop() {
		$mtime = microtime();
		$mtime = explode(' ', $mtime);
		$time_end = $mtime[1] + $mtime[0];
		$time_total = $time_end - $this->time_start;
		return $time_total;
	}

	/**
	 * Wraps fatal errors in a nice header and footer and dies.
	 * @param string $message
	 */
	function bail($message) { // Just wraps errors in a nice header and footer
		if ( !$this->show_errors ) {
			return false;
		}
		else {
			show_error('1', $message);
		}
	}
}
?>